package com.welse.modulebiz.utils;

import com.auth0.jwt.JWT;
import com.auth0.jwt.JWTVerifier;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.exceptions.JWTVerificationException;
import com.auth0.jwt.interfaces.DecodedJWT;

import java.util.Date;
import java.util.Map;

public class JwtUtil {

    private static final String SECRET = "welse-thn-module-biz-secret-key";
    private static final long EXPIRATION = 1000 * 60 * 60 * 24 * 7; // 7天

    /**
     * 生成JWT令牌
     *
     * @param claims 要包含在令牌中的声明
     * @return JWT令牌字符串
     */
    public static String generateToken(Map<String, Object> claims) {
        Date expireDate = new Date(System.currentTimeMillis() + EXPIRATION);

        return JWT.create()
                .withClaim("claims", claims)
                .withExpiresAt(expireDate)
                .sign(Algorithm.HMAC256(SECRET));
    }

    /**
     * 验证JWT令牌
     *
     * @param token JWT令牌
     * @return 解码后的JWT对象
     * @throws JWTVerificationException 当令牌无效时抛出异常
     */
    public static DecodedJWT verifyToken(String token) throws JWTVerificationException {
        JWTVerifier verifier = JWT.require(Algorithm.HMAC256(SECRET)).build();
        return verifier.verify(token);
    }

    /**
     * 解析JWT令牌中的声明
     *
     * @param token JWT令牌
     * @return 令牌中的声明映射
     */
    public static Map<String, Object> parseToken(String token) {
        return JWT.require(Algorithm.HMAC256(SECRET))
                .build()
                .verify(token)
                .getClaim("claims")
                .asMap();
    }

    /**
     * 检查JWT令牌是否过期
     *
     * @param token JWT令牌
     * @return 如果令牌已过期返回true，否则返回false
     */
    public static boolean isTokenExpired(String token) {
        try {
            DecodedJWT jwt = JWT.require(Algorithm.HMAC256(SECRET))
                    .build()
                    .verify(token);
            return jwt.getExpiresAt().before(new Date());
        } catch (Exception e) {
            return true; // 如果验证失败，认为令牌已过期
        }
    }

    /**
     * 刷新JWT令牌
     *
     * @param token 原始JWT令牌
     * @return 新的JWT令牌
     */
    public static String refreshToken(String token) {
        Map<String, Object> claims = parseToken(token);
        return generateToken(claims);
    }
}
